working pyspark with json